/*
    基础工具包：其他功能模块共用的工具
    时间、字符串、文件相关工具
*/

#pragma once
#include <iostream>
#include <string>
#include <sys/stat.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <vector>
#include <fcntl.h>
#include <time.h>
#include <atomic> //提供了具有原子性的计数器
#include <unistd.h>
#include <fstream>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    const std::string tempfile_path = "./tempfile/"; //产生的所有临时文件所在路径

    class TimeUtil
    {
    public:
        //获取时间字符串，0：毫秒级时间戳字符串，1：UTC时间字符串
        static std::string getTimeStr(int opt)
        {
            struct timeval tv;
            gettimeofday(&tv, nullptr);
            if(opt == 0)
            {
                return std::to_string(tv.tv_sec * 1000 + tv.tv_usec / 1000);
            }
            else
            {
                std::string curtime = std::string(ctime(&tv.tv_sec));
                curtime[curtime.size()-1] = '\0'; //消除最后的换行符
                return curtime;
            }
        }
    };

    class PathUtil
    {
    public:
        static std::string addPath(const std::string file)
        {
            return (tempfile_path + file);
        }
    };

    class FileUtil
    {
    public:
        // 根据毫秒级时间戳+原子性计数器形成唯一文件名（不具有后缀）
        static std::string uniqFilename()
        {
            static std::atomic_uint id(0);
            id++;
            std::string filename = TimeUtil::getTimeStr(0);
            filename += "_" + to_string(id);
            return filename;
        }

        // 源文件添加后缀（filename+后缀）
        static std::string addSuffixOrg(const std::string& filename)
        {
            return filename + ".cc";
        }
        
        // 可执行文件添加后缀（filename+后缀）
        static std::string addSuffixExe(const std::string& filename)
        {
            return filename + ".exe";
        }

        // 编译错误文件添加后缀（filename+后缀）
        static std::string addSuffixComperr(const std::string& filename)
        {
            return filename + ".compile_err";
        }

        // 标准输入文件添加后缀（filename+后缀）
        static std::string addSuffixStdin(const std::string& filename)
        {
            return filename + ".stdin";
        }

        // 标准输出文件添加后缀（filename+后缀）
        static std::string addSuffixStdout(const std::string& filename)
        {
            return filename + ".stdout";
        }

        // 标准错误文件添加后缀（filename+后缀）
        static std::string addSuffixStderr(const std::string& filename)
        {
            return filename + ".stderr";
        }

        // 判断一个文件是否存在
        static bool isFileExist(const std::string pathname)
        {
            struct stat st;
            //获取文件属性，如果成功获取，则文件存在
            return (0 == stat(pathname.c_str(), &st));
        }

        // 写文件
        static bool writeFile(const std::string filepath, const std::string& code)
        {
            std::ofstream out(filepath);
            if(!out.is_open())
                return false;
            out.write(code.c_str(), code.size());
            out.close();
            return true;
        }

        // 读文件
        static bool readFile(std::string filepath, std::string& code)
        {
            code.clear();
            std::ifstream in(filepath);
            if(!in.is_open())
                return false;
            
            std::string line;
            while(std::getline(in, line))
            {
                code += line;
                code += "\n";
            }
            in.close();
            return true;
        }

        //清理所有临时文件
        static void clearTempfile(std::string& filename)
        {
            std::vector<std::string> pathnamev;
            pathnamev.emplace_back(PathUtil::addPath(FileUtil::addSuffixOrg(filename))); //源文件
            pathnamev.emplace_back(PathUtil::addPath(FileUtil::addSuffixExe(filename))); //可执行
            pathnamev.emplace_back(PathUtil::addPath(FileUtil::addSuffixComperr(filename))); //编译报错
            pathnamev.emplace_back(PathUtil::addPath(FileUtil::addSuffixStdin(filename))); //标准输入
            pathnamev.emplace_back(PathUtil::addPath(FileUtil::addSuffixStdout(filename))); //标准输出
            pathnamev.emplace_back(PathUtil::addPath(FileUtil::addSuffixStderr(filename))); //标准错误
            for(auto& e : pathnamev)
                if(FileUtil::isFileExist(e))
                    unlink(e.c_str());
        }
    };

    class StrUtil
    {
    public:
        static void splitStr(const std::string& s, std::vector<std::string>& tokens, const std::string& sep)
        {
            boost::split(tokens, s, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };
}